package com.zhangh.summer.admin.support.utils;

import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.HorizontalAlignment;

import java.util.List;

/**
 * @description: TODO
 * @author: zhanghui  zhanghlucky@163.com
 * @create: 2021-09-10 23:12
 **/
public class ExcelUtils {
    /**
     *
     * @param sheetName
     * @param titleList
     * @param valueList
     * @param wb
     * @return
     */
    public static HSSFWorkbook getHSSFWorkbook(String sheetName, List<String> titleList, List<List<String>> valueList, HSSFWorkbook wb){
        // 第一步，创建一个HSSFWorkbook，对应一个Excel文件
        if(wb == null){
            wb = new HSSFWorkbook();
        }
        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet(sheetName);

        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
        HSSFRow row = sheet.createRow(0);

        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HorizontalAlignment.CENTER); // 创建一个居中格式

        //声明列对象
        HSSFCell cell = null;

        //创建标题
        for(int i=0;i<titleList.size();i++){
            cell = row.createCell(i);
            cell.setCellValue(titleList.get(i));
            cell.setCellStyle(style);
        }

        //创建内容
        for(int i=0;i<valueList.size();i++){
            row = sheet.createRow(i + 1);
            List<String> dataList=valueList.get(i);
            for(int j=0;j<dataList.size();j++){
                //将内容按顺序赋给对应的列对象
                //row.createCell(j).setCellValue(values[i][j]);
                String data=dataList.get(j);
                row.createCell(j).setCellValue(data);
            }
        }
        return wb;
    }
}
